package proxy.modules;

/**
 * @author Feng
 * @date 2019/3/12 20:49
 */
public class DynamicDataSourceEntry {
    // 默认数据源
    public final static String DEFALT_DATASOURCE = null;

    private final static ThreadLocal<String> local = new ThreadLocal<>();

    private DynamicDataSourceEntry(){}

    /**
     * 清空数据源
     */
    public static void clear(){
        local.remove();
    }

    /**
     * 获取当前正在使用的数据源
     * @return
     */
    public static String get(){
        return  local.get();
    }

    /**
     * 还原当前切面的数据源
     */
    public static void restore(){
        local.set(DEFALT_DATASOURCE);
    }

    /**
     * 设置已知名字的数据源
     * @param resource
     */
    public static void set(String resource){
        local.set(resource);
    }

    /**
     * 根据年份自动设置数据源
     * @param year
     */
    public static void set(int year){
        local.set("DB_" + year);
    }

}
